updated requirements

app.py

@@ -12,9 +12,6 @@ MODELS={
                 'tokenizer':AutoTokenizer.from_pretrained("jhonparra18/petro-twitter-assistant-30ep-large"),
                 'model':AutoModelForCausalLM.from_pretrained("jhonparra18/petro-twitter-assistant-30ep-large")}}
 
-# MODELS={'petro':pipeline('text-generation', model='jhonparra18/petro-twitter-assistant-30ep-large'),
-#         'uribe':pipeline('text-generation', model='jhonparra18/uribe-twitter-assistant-30ep')}
-
 
 def text_completion(tokenizer,model,input_text:str,max_len:int=100):
     tokenizer.padding_side="left" ##start padding from left to right

requirements.txt

@@ -1,6 +1,5 @@
-pandas
-datasets
-transformers
-wandb
-torch
-streamlit
+pandas==1.5.3
+datasets==2.9.0
+transformers==4.26.1
+torch==1.13.1
+streamlit==1.18.1