sivapriya14 commited on
Commit
be5f34e
·
verified ·
1 Parent(s): 443e84d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -3,8 +3,11 @@ import os
3
  from flask import Flask, request, jsonify
4
  from transformers import pipeline
5
 
6
- # Load a public model that doesn't require authentication
7
- llm = pipeline("text-generation", model="tiiuae/falcon-7b-instruct")
 
 
 
8
 
9
  # Load the schemes data from JSON if available
10
  SCHEMES_FILE = "schemes.json"
 
3
  from flask import Flask, request, jsonify
4
  from transformers import pipeline
5
 
6
+ from transformers import pipeline
7
+
8
+ llm = pipeline("text-generation", model="tiiuae/falcon-1b-instruct") # 1B model instead of 7B
9
+
10
+
11
 
12
  # Load the schemes data from JSON if available
13
  SCHEMES_FILE = "schemes.json"